package com.lagou.base;

import com.alibaba.fastjson.JSON;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Map;

public class BaseServlet extends HttpServlet {

    /**
     * doGet方法作为一个调度器,根据请求功能的不同,调用对应的方法
     *      规定必须传一个参数
     *              methodName=功能名
     */
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取参数 要访问的方法名
      //  String methodName = req.getParameter("methodName");
        String methodName = null;
        //1.获取post请求的Content-Type类型
        String contentType = req.getHeader("Content-Type");
        //2.判断传递的参数是不是JSON类型
            if("application/json;charset=utf-8".equalsIgnoreCase(contentType)){
                String postJSON = getPostJSON(req);
                //将JSON格式的字符串转换为map
                Map<String,Object> map = JSON.parseObject(postJSON, Map.class);
                //从map中获取methodName
                methodName = (String) map.get("methodName");
                req.setAttribute("map",map);

            }else{
                methodName = req.getParameter("methodName");
            }
        //2.判断 执行的方法

        if(null!=methodName){
            //通过反射优化代码 提升代码的可维护性

            try {
                //1.获取字节码文件对象
                Class c  = this.getClass();
                //2.根据传人的方法名,获取对应的方法对象
                Method method = c.getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);
                //3.调用method对象的 invoke方法执行对应的功能
                method.invoke(this,req,resp);
            } catch (NoSuchMethodException e) {
                e.printStackTrace();
            } catch (Exception e) {
                e.printStackTrace();
                System.out.println("请求的功能不存在");
            }

        }
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        doGet(req, resp);
    }/**
     * POST请求格式为 application/json;charset=utf-8
     * 在这个方法中我们使用流的方式,获取到POST请求的数据
     * */
    public String getPostJSON(HttpServletRequest request){
        try {
            //1.从request中 获取缓冲输入流
            BufferedReader reader = request.getReader();
            //2.创建StringBuffer 保存读取的数据
            StringBuffer sb = new StringBuffer();
            String line = null;
            //3.循环读取
            while((line = reader.readLine())!=null){
                sb.append(line);
            }
            return  sb.toString();
            //4.返回结果
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }

}
